# -*- encoding: utf-8 -*-
"""

PyCharm 07视频检测
2024年10月20日17时19分
by wenyang

"""
import cv2 as cv
# 检测函数
def face_detect_demo(img):
    # 首先将图片转化为一个灰度图像
    gray_img = cv.cvtColor(img, cv.COLOR_BGR2GRAY)
    # 加载人脸识别分类器 就是已经训练好的模型进行使用1
    face_detect = cv.CascadeClassifier("E:/DOWMLOAD/chrome_download/opencv/sources/data/haarcascades/haarcascade_frontalface_default.xml")
    face = face_detect.detectMultiScale(gray_img, scaleFactor = 1.1)
    for x, y, w, h in face:
        cv.rectangle(img, (x, y), (x + w, y + h), color = (0, 0, 255), thickness = 2)
    cv.imshow("result", img)


def main():
    # 读取摄像头
    cap = cv.VideoCapture("1.mp4")
    while True:
        flag, frame = cap.read()
        if not flag:
            break
        face_detect_demo(frame)
        if ord("q") == cv.waitKey(1): # 此时要设置为1，因为视频的帧率是1
            break
    # 释放内存
    cv.destroyAllWindows()
    # 释放摄像头
    cap.release()



if __name__ == "__main__":
    main()
